7f5ca0651646f7cee286609e8c90f6984686a356,test-framework/performance/src/main/java/com/continuuity/performance/runner/PerformanceTestRunner.java,PerformanceTestRunner,init,#,280
Before Change
}
Module dataFabricModule;
if (config.get("perf.datafabric.mode") != null
&& config.get("perf.datafabric.mode").equals("distributed")) {
dataFabricModule = new DataFabricModules().getDistributedModules();
} else {
dataFabricModule = new DataFabricModules().getSingleNodeModules();
}
try {
injector = Guice
.createInjector(dataFabricModule,
new ConfigModule(config),
new IOModule(),
new LocationRuntimeModule().getInMemoryModules(),
new DiscoveryRuntimeModule().getInMemoryModules(),
new ProgramRunnerRuntimeModule().getInMemoryModules(),
new MetricsClientRuntimeModule().getInMemoryModules(),
After Change
Module metricsClientModule;
try {
if (config.get("perf.reactor.mode") != null
&& config.get("perf.reactor.mode").equals("distributed")) {
dataFabricModule = new DataFabricModules().getDistributedModules();
locationModule = new LocationRuntimeModule().getDistributedModules();
programRunnerModule = new ProgramRunnerRuntimeModule().getDistributedModules();
metricsClientModule = new MetricsClientRuntimeModule().getInMemoryModules();
zkClientService =
ZKClientServices.delegate(
ZKClients.reWatchOnExpire(
ZKClients.retryOnFailure(
ZKClientService.Builder.of(
config.get(Constants.CFG_ZOOKEEPER_ENSEMBLE)).setSessionTimeout(10000).build(),
RetryStrategies.fixDelay(2, TimeUnit.SECONDS))));
discoveryServiceModule = new DiscoveryRuntimeModule(zkClientService).getDistributedModules();
} else {
dataFabricModule = new DataFabricModules().getSingleNodeModules();
locationModule = new LocationRuntimeModule().getInMemoryModules();
programRunnerModule = new ProgramRunnerRuntimeModule().getInMemoryModules();
metricsClientModule = new MetricsClientRuntimeModule().getInMemoryModules();
discoveryServiceModule = new AbstractModule() {